package com.demo.CodeAutoGenerator;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import java.util.ArrayList;

//这个类是MP的逆向工程,可以自动生成代码,这是一个工具类可以拿来进行修改后使用

//需要修改的地方: 第60行的映射表名,以及一些值,主要是第60行

//这是一个自动代码生成器
public class CodeGenerator {
    public static void main(String[] args) {
      //需要构建一个 代码自动生成器 对象
        AutoGenerator mpg=new AutoGenerator();

        //1.全局配置  这里的GlobalConfig不能导错包
        GlobalConfig gc=new GlobalConfig();
        String property = System.getProperty("user.dir");//得到项目的路径
        gc.setOutputDir(property+"/src/main/java");
        gc.setAuthor("鱼骨头鱼骨头");
        gc.setOpen(false);//就是运行之后需不需要打开文件夹 == win+e
        gc.setFileOverride(false);//是否覆盖
        gc.setServiceName("%sService");  // 去掉Servcie的I前缀
        gc.setIdType(IdType.ID_WORKER);
        gc.setDateType(DateType.ONLY_DATE); //日期类型
        gc.setSwagger2(true);
        mpg.setGlobalConfig(gc);

        //2. 设置数据源
        DataSourceConfig dsc=new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/mybatis_plus?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2B8\n");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setPassword("root118");
        dsc.setUsername("root");
        dsc.setDbType(DbType.MYSQL); //设置数据库的类型
        mpg.setDataSource(dsc);

        //3.包的配置
        PackageConfig pc=new PackageConfig();
        pc.setModuleName("blog"); //一会的下面的service,controller,mapper类的代码都会在blog的包下
        pc.setParent("com.demo");
        pc.setEntity("pojo"); //设置实体类的包名
        pc.setMapper("mapper");
        pc.setService("service");
        pc.setController("controller");
        mpg.setPackageInfo(pc);

        //4.策略配置
        StrategyConfig strategy=new StrategyConfig();
        strategy.setInclude("user"); //参数为设置需要映射的表名,可以同时设置多个,会生成实体类
        strategy.setNaming(NamingStrategy.underline_to_camel); //让包名能从下户线转驼峰命名
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);//让字段名能从下户线转驼峰命名
        strategy.setEntityLombokModel(true);  //自动Lombok
        strategy.setLogicDeleteFieldName("delete"); //设置逻辑删除的字段

        //自动填充配置
        TableFill gmtCreate=new TableFill("gmt_Create", FieldFill.INSERT); //设置字段的填充
        TableFill gmtModified=new TableFill("gmt_modified", FieldFill.UPDATE);
        ArrayList<TableFill> tableFills=new ArrayList<>();
        tableFills.add(gmtCreate);
        tableFills.add(gmtModified);
        strategy.setTableFillList(tableFills);

        //乐观锁配置
        strategy.setVersionFieldName("version");
        mpg.setStrategy(strategy);
        mpg.execute();//执行代码

    }

}
